# -*- coding: utf-8 -*-
"""
Created on Thu Oct  8 23:05:14 2020

@author: 郑歆羽

等差数列计算小程序

Version 1.0

"""


step=eval(input("请输入公差:"))
number=eval(input('请输入项数:'))


print('请问您是选择求首项还是末项,亦或求数列和？')
state0=input('请输入选择结果（求首项或求末项或求数列和）:')


if state0=='求末项':
    begin=eval(input('请输入首项:'))
    result=end=begin+(number-1)*step
    print(result)
    

elif state0=='求首项':
    end=eval(input('请输入末项:'))
    result=begin=end-step*(number-1)
    print(result)


elif state0=='求数列和':
    print('请输入您所拥有的信息。')
    print('请问您拥有首项还是末项，亦或是两项都有？')
    state2=input('请输入选择结果(首项或末项或两项都有):')
    
    if state2=='首项':
        begin=eval(input('请输入首项:'))
        end=begin+(number-1)*step
        result=sum=(begin+end)*number/2
        result=sum
        print(result)
    
    elif state2=='末项':
        end=eval(input('请输入末项:'))
        begin=end-step*(number-1)
        result=sum=(begin+end)*number/2
        print(result)
        
    elif state2=='两项都有':
        begin=eval(input('请输入首项:'))
        end=eval(input('请输入末项:'))
        result=sum=(begin+end)*number/2
        print(result)     
    
    else:
        print('您输入的信息有误，请重新输入！')   


else:
    print('您输入的信息有误，请重新输入！')